import tiktoken

import os

from langchain_openai import ChatOpenAI

os.environ["DASHSCOPE_API_KEY"] = 'sk-c44402d7a12c41299bb716af8d7e8bac'

llm = ChatOpenAI(
    api_key=os.getenv("DASHSCOPE_API_KEY"),  # 如果您没有配置环境变量，请在此处用您的API Key进行替换
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",  # 填写DashScope base_url
    model="qwen-plus"
)

tiktoken.encoding_for_model("gpt-3.5-turbo")

wb_prompt_template = """
    (C) 上下文:
    我是一个测评过很多数码产品的微博大V，通常会分享与科技产品相关的内容，特别是手机测评。最近，一个知名视频博主发布了一篇详细的手机测评报告，你需要根据这篇报告撰写一个微博文案，以便向你的粉丝推荐这款手机并提供你的见解。

    手机测评报告: {phone_name}

    (O) 目的:
    撰写一个吸引人的微博文案，概要介绍视频博主的手机测评报告，并提供你的简短评价，鼓励粉丝观看原视频。

    (S) 样式:
    使用科技测评领域常见的专业且亲切的风格，展示你对手机测评的专业知识，同时保持轻松易懂。

    (T) 语调:
    采用热情且可信赖的语气，传递你对这款手机的真实评价，让粉丝感受到你的专业性和热情。

    (A) 受众:
    目标受众为对科技产品和手机测评感兴趣的普通消费者，他们希望了解最新手机的特点和性能。

    (R) 响应:
    文微博文案的格式应包括:简要介绍、主要亮点、个人评价、鼓励粉丝去购买这个手机的呼吁。保持在140字左右，以适应微博平台的限制。
"""

token = encoding.encode(wb_prompt_template)

print(token)

print(len(token))